/*
Copyright 2017 HugeGraph Authors

Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with this
work for additional information regarding copyright ownership. The ASF
licenses this file to You under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
License for the specific language governing permissions and limitations
under the License.
*/

package v1

import (
    corev1 "k8s.io/api/core/v1"
    "k8s.io/apimachinery/pkg/api/resource"
    metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// EDIT THIS FILE!  THIS IS SCAFFOLDING FOR YOU TO OWN!
// NOTE: json tags are required.
// Any new fields you add must have json tags for the fields to be serialized.

// ComputerJobSpec defines the desired state of HugeGraphComputerJob
type ComputerJobSpec struct {
    AlgorithmName *string `json:"algorithmName"`

    JobId *string `json:"jobId"`

    Image *string `json:"image"`

    // Image pull policy. One of Always, Never, IfNotPresent. Defaults to Always
    // if :latest tag is specified, or IfNotPresent otherwise.
    // +kubebuilder:validation:Enum=Always;Never;IfNotPresent
    PullPolicy corev1.PullPolicy `json:"pullPolicy,omitempty"`

    // Secrets for image pull.
    PullSecrets []corev1.LocalObjectReference `json:"pullSecrets,omitempty"`

    //+kubebuilder:validation:Minimum=1
    WorkerInstances int32 `json:"workerInstances"`

    MasterCpu resource.Quantity `json:"masterCpu,omitempty"`

    WorkerCpu resource.Quantity `json:"workerCpu,omitempty"`

    MasterMemory resource.Quantity `json:"masterMemory,omitempty"`

    WorkerMemory resource.Quantity `json:"workerMemory,omitempty"`

    ComputerConf map[string]string `json:"computerConf"`

    Log4jXml *string `json:"log4jXml,omitempty"`
    
    JarFile *string `json:"jarFile,omitempty"`

    //+kubebuilder:validation:Pattern=`^(http|https):\/\/([\w.]+\/?)\S*$`
    RemoteJarUri *string  `json:"remoteJarUri,omitempty"`

    JvmOptions *string `json:"jvmOptions,omitempty"`

    // Environment variables shared by all Master and Worker.
    EnvVars []corev1.EnvVar `json:"envVars,omitempty"`

    // Environment variables injected from a source, shared by all Master and Worker.
    EnvFrom []corev1.EnvFromSource `json:"envFrom,omitempty"`

    MasterCommand []string `json:"masterCommand,omitempty"`

    MasterArgs []string `json:"masterArgs,omitempty"`

    WorkerCommand []string `json:"workerCommand,omitempty"`

    WorkerArgs []string `json:"workerArgs,omitempty"`

    // Volumes in the Job pod.
    // More info: https://kubernetes.io/docs/concepts/storage/volumes/
    Volumes []corev1.Volume `json:"volumes,omitempty"`

    // Volume mounts in the Job container.
    // More info: https://kubernetes.io/docs/concepts/storage/volumes/
    VolumeMounts []corev1.VolumeMount `json:"volumeMounts,omitempty"`

    SecretPaths map[string]string `json:"secretPaths,omitempty"`

    ConfigMapPaths map[string]string `json:"configMapPaths,omitempty"`

    //+kubebuilder:validation:Schemaless
    //+kubebuilder:pruning:PreserveUnknownFields
    // More info: https://kubernetes.io/docs/concepts/workloads/pods/#pod-templates
    PodTemplateSpec corev1.PodTemplateSpec `json:"podTemplateSpec,omitempty"`

    // More info: https://kubernetes.io/zh/docs/tasks/configure-pod-container/security-context/
    SecurityContext corev1.SecurityContext `json:"securityContext,omitempty"`
}

type ComputerJobState struct {
    //+optional
    Superstep int32 `json:"superstep"`

    //+optional
    MaxSuperstep int32 `json:"maxSuperstep"`

    //+optional
    LastSuperstepStat *string `json:"lastSuperstepStat"`
}

// ComputerJobStatus defines the observed state of HugeGraphComputerJob
type ComputerJobStatus struct {
    JobStatus *string `json:"jobStatus"`

    //+optional
    JobState ComputerJobState `json:"jobState"`

    //+optional
    Components ComponentsStatus `json:"componentStates"`

    LastUpdateTime metav1.Time `json:"lastUpdateTime,omitempty"`
}

type ComponentState struct {
    // The resource name of the component.
    Name string `json:"name"`

    // The state of the component.
    State string `json:"state"`

    //+optional
    Message string `json:"message"`

    //+optional
    ErrorLog string `json:"errorLog"`
}

type ComponentsStatus struct {
    // +optional
    ConfigMap ComponentState `json:"configMap"`

    // +optional
    MasterJob ComponentState `json:"masterJob"`

    // +optional
    WorkerJob ComponentState `json:"workerJob"`
}

// +kubebuilder:object:root=true
// +kubebuilder:subresource:status
// +kubebuilder:resource:shortName=hcjob
// +kubebuilder:printcolumn:JSONPath=".spec.jobId",name=JobId,type=string
// +kubebuilder:printcolumn:JSONPath=".status.jobStatus",name=JobStatus,type=string
// +kubebuilder:printcolumn:JSONPath=".status.jobState.superstep",name=Superstep,type=integer
// +kubebuilder:printcolumn:JSONPath=".status.jobState.maxSuperstep",name=MaxSuperstep,type=integer
// +kubebuilder:printcolumn:JSONPath=".status.jobState.lastSuperstepStat",name=SuperstepStat,type=string

// HugeGraphComputerJob is the Schema for the hugegraphcomputerjobs API
type HugeGraphComputerJob struct {
    metav1.TypeMeta   `json:",inline"`
    metav1.ObjectMeta `json:"metadata,omitempty"`

    Spec   ComputerJobSpec   `json:"spec,omitempty"`
    Status ComputerJobStatus `json:"status,omitempty"`
}

// +kubebuilder:object:root=true

// HugeGraphComputerJobList contains a list of HugeGraphComputerJob
type HugeGraphComputerJobList struct {
    metav1.TypeMeta `json:",inline"`
    metav1.ListMeta `json:"metadata,omitempty"`

    Items []HugeGraphComputerJob `json:"items"`
}

func init() {
    SchemeBuilder.Register(&HugeGraphComputerJob{}, &HugeGraphComputerJobList{})
}
